import asyncio
from enum import Enum
from xhs_auto_poster.common.logger_util import get_logger

logger = get_logger(__name__)

# 初始化消息队列
logger.info("初始化异步消息队列")
msg_queue: asyncio.Queue = asyncio.Queue()


class MSG_TYPE(Enum):
    """消息类型枚举类"""
    THINK = 1          # 思考过程
    DONE = 2           # 完成标识
    ANSWER = 3         # 最终答案
    THINK_CHUNK = 4    # 思考过程片段
    QRCODE = 5         # 二维码
    QXQRCODE = 6       # 企业微信二维码


# 测试
if __name__ == "__main__":
    logger.debug(f"消息类型测试: {MSG_TYPE.THINK.name}")
    print(MSG_TYPE.THINK.name)
